#if defined(UNICODE) && !defined(_UNICODE)
    #define _UNICODE
#elif defined(_UNICODE) && !defined(UNICODE)
    #define UNICODE
#endif
#include "CSolver.h"
#define MAIN_ICON 2000
/*  Declare Windows procedure  */
LRESULT CALLBACK WindowProcedure (HWND, UINT, WPARAM, LPARAM);

/*  Make the class name into a global variable  */
TCHAR szClassName[ ] = _T("CodeBlocksWindowsApp");
CCanvas *lpCanvas;
int WINAPI WinMain (HINSTANCE hThisInstance,
                     HINSTANCE hPrevInstance,
                     LPSTR lpszArgument,
                     int nCmdShow)
{
    HWND hwnd;               /* This is the handle for our window */
    MSG messages;            /* Here messages to the application are saved */
    WNDCLASSEX wincl;        /* Data structure for the windowclass */

    /* The Window structure */
    wincl.hInstance = hThisInstance;
    wincl.lpszClassName = szClassName;
    wincl.lpfnWndProc = WindowProcedure;      /* This function is called by windows */
    wincl.style = CS_DBLCLKS;                 /* Catch double-clicks */
    wincl.cbSize = sizeof (WNDCLASSEX);

    /* Use default icon and mouse-pointer */
    wincl.hIcon = LoadIcon(NULL, MAKEINTRESOURCE(MAIN_ICON));
    wincl.hIconSm = LoadIcon (NULL, MAKEINTRESOURCE(MAIN_ICON));
    wincl.hCursor = LoadCursor (NULL, IDC_ARROW);
    wincl.lpszMenuName = NULL;                 /* No menu */
    wincl.cbClsExtra = 0;                      /* No extra bytes after the window class */
    wincl.cbWndExtra = 0;                      /* structure or the window instance */
    /* Use Windows's default colour as the background of the window */
    wincl.hbrBackground = (HBRUSH) COLOR_BACKGROUND;

    /* Register the window class, and if it fails quit the program */
    if (!RegisterClassEx (&wincl))
        return 0;
    CCanvas canvas;
    lpCanvas = &canvas;
    /* The class is registered, let's create the program*/
    hwnd = CreateWindowEx (
           0,                   /* Extended possibilites for variation */
           szClassName,         /* Classname */
           _T("Code::Blocks Template Windows App"),       /* Title Text */
           WS_OVERLAPPEDWINDOW, /* default window */
           CW_USEDEFAULT,       /* Windows decides the position */
           CW_USEDEFAULT,       /* where the window ends up on the screen */
           600,                 /* The programs width */
           550,                 /* and height in pixels */
           HWND_DESKTOP,        /* The window is a child-window to desktop */
           NULL,                /* No menu */
           hThisInstance,       /* Program Instance handler */
           NULL                 /* No Window Creation data */
           );

    /* Make the window visible on the screen */
    ShowWindow (hwnd, nCmdShow);

    /* Run the message loop. It will run until GetMessage() returns 0 */
    while (GetMessage (&messages, NULL, 0, 0))
    {
        /* Translate virtual-key messages into character messages */
        TranslateMessage(&messages);
        /* Send message to WindowProcedure */
        DispatchMessage(&messages);
    }

    /* The program return-value is 0 - The value that PostQuitMessage() gave */
    return messages.wParam;
}


/*  This function is called by the Windows function DispatchMessage()  */

LRESULT CALLBACK WindowProcedure (HWND hwnd, UINT message, WPARAM wParam, LPARAM lParam)
{
    PAINTSTRUCT ps;
    HDC hdc;
    int x, y;
    switch (message)                  /* handle the messages */
    {
    case WM_KEYUP:
        switch (wParam)
        {
        case 'R':
            if (!lpCanvas->m_klotski.m_bSolving)
            {
                lpCanvas->m_klotski.m_fnRestart();
            }
            break;
        case 'S':
            if (lpCanvas->m_klotski.m_bSolving)
            {
                lpCanvas->m_klotski.m_bSolving = false;
                KillTimer(hwnd, 1000);
            }
            else if (!lpCanvas->m_bEditing)
            {
                fnSolve(lpCanvas);
                SetTimer(hwnd, 1000, 40, NULL);
            }
            break;
        case '0':
            lpCanvas->m_klotski.m_iStep = 0;
            strcpy(lpCanvas->m_klotski.m_lpbtData, lpCanvas->m_klotski.m_lpbtStart);
            break;
        case '1':
        case '2':
        case '3':
        case '4':
        case '5':
        case '6':
        case '7':
        case '8':
        case '9':
            lpCanvas->m_klotski.m_fnGame(int(wParam - '0'));
            lpCanvas->m_klotski.m_fnRestart();
            break;
        case 'E':
            if (!lpCanvas->m_klotski.m_bSolving)
            {
                if (lpCanvas->m_bEditing)
                {
                    strcpy(lpCanvas->m_klotski.m_lpbtStart, lpCanvas->m_klotski.m_lpbtData);
                    lpCanvas->m_bEditing = false;
                }
                else
                {
                    lpCanvas->m_bEditing = true;
                }
            }
            break;
        case VK_LEFT:
            if (lpCanvas->m_klotski.m_iStep > 0)
            {
                --(lpCanvas->m_klotski.m_iStep);
                strcpy(lpCanvas->m_klotski.m_lpbtData, lpCanvas->m_klotski.m_vecSolution[lpCanvas->m_klotski.m_iStep].m_lpbtData);
            }
            break;
        case VK_RIGHT:
            if (lpCanvas->m_klotski.m_iStep < int(lpCanvas->m_klotski.m_vecSolution.size()) - 1)
            {
                ++(lpCanvas->m_klotski.m_iStep);
                strcpy(lpCanvas->m_klotski.m_lpbtData, lpCanvas->m_klotski.m_vecSolution[lpCanvas->m_klotski.m_iStep].m_lpbtData);
            }
            break;
        }
        InvalidateRect(hwnd, NULL, true);
        break;
    case WM_LBUTTONUP:
        x = LOWORD(lParam);
        y = HIWORD(lParam);
        lpCanvas->OnMouseEvent(x, y);
        InvalidateRect(hwnd, NULL, true);
        break;
    case WM_TIMER:
        if (!lpCanvas->m_klotski.m_bSolving)
        {
            KillTimer(hwnd, 1000);
        }
        InvalidateRect(hwnd, NULL, true);
        break;
    case WM_PAINT:
        hdc = BeginPaint(hwnd, &ps);
        lpCanvas->OnPaint(hdc);
        EndPaint(hwnd, &ps);
        break;
    case WM_DESTROY:
        lpCanvas->m_klotski.m_bSolving = false;
        KillTimer(hwnd, 1000);
        PostQuitMessage (0);       /* send a WM_QUIT to the message queue */
        break;
    default:                      /* for messages that we don't deal with */
        return DefWindowProc (hwnd, message, wParam, lParam);
    }

    return 0;
}
